package edu.br.domain;

import java.util.List;

import edu.br.enuns.Tag;

public class TrainingPlan {

	// 6 trainings max
	private List<Training> training;

	private Tag lastTag;

	public Tag nextTag() {
		Tag nextTag = null;
		if (this.lastTag == null) {
			nextTag = Tag.A;
		} else if (this.lastTag.equals(Tag.A)) {
			nextTag = Tag.B;
		} else if (this.lastTag.equals(Tag.B)) {
			nextTag = Tag.C;
		} else if (this.lastTag.equals(Tag.C)) {
			nextTag = Tag.D;
		} else if (this.lastTag.equals(Tag.D)) {
			nextTag = Tag.E;
		} else if (this.lastTag.equals(Tag.E)) {
			nextTag = Tag.F;
		}
		return nextTag;
	}

	public List<Training> getTraining() {
		return this.training;
	}

	public void setTraining(List<Training> training) {
		this.training = training;
	}

	public void addTraining(Training training) {
		this.training.add(training);
	}

	public void removeTraining(Training training) {
		this.training.remove(training);
	}

	public Tag getlastTag() {
		return this.lastTag;
	}

	public void setLastTag(Tag tag) {
		this.lastTag = tag;
	}
}